source activate ke2torch23cu121

# llama2-7b-chat
model=llama2-7b-chat
method=ICE
date=2025-03-26

data_dir=./dataset/ICE/wikidata_recent.json
datatype=recent
# data_dir=./dataset/ICE/zsre.json
# datatype=zsre
# data_dir=./dataset/ICE/wikidata_counterfact.json
# datatype=counterfact
# data_dir=./dataset/ICE/wikibio.json
# datatype=wikibio
method_type=js
cnt=2
echo ${date}/${method}-${method_type}/${model}-${datatype}-${cnt}
CUDA_VISIBLE_DEVICES=2 python examples/run_knowedit_llama2.py \
    --editing_method=${method} \
    --hparams_dir=./hparams/${method}/${model}-${method_type}.yaml \
    --data_dir=${data_dir} \
    --datatype=${datatype} \
    --metrics_save_dir=./results/${model}/${method}-${method_type}-${date} \
    --pre_file=../EasyEdit/pre_edit/${model}_${datatype}_pre_edit.json \
    > log/${date}/${method}-${method_type}/${model}-${datatype}-${cnt}.log 2>&1 &

method_type=kl
cnt=3
echo ${date}/${method}-${method_type}/${model}-${datatype}-${cnt}
CUDA_VISIBLE_DEVICES=3 python examples/run_knowedit_llama2.py \
    --editing_method=${method} \
    --hparams_dir=./hparams/${method}/${model}-${method_type}.yaml \
    --data_dir=${data_dir} \
    --datatype=${datatype} \
    --metrics_save_dir=./results/${model}/${method}-${method_type}-${date} \
    --pre_file=../EasyEdit/pre_edit/${model}_${datatype}_pre_edit.json \
    > log/${date}/${method}-${method_type}/${model}-${datatype}-${cnt}.log 2>&1 &